from typing import List

from pydantic import BaseModel

from server.framework.utils.kubernetes.schemas.pod_info import PodInfo


class NodeInfo(BaseModel):
    labels: dict
    node_name: str
    gpu: int
    docker_version: str
    kernel_version: str
    os_image: str
    ip: str
    pods: List[PodInfo]
